/*
 * Copyright © 2018 www.noark.xyz All Rights Reserved.
 *
 * 感谢您选择Noark框架，希望我们的努力能为您提供一个简单、易用、稳定的服务器端框架 ！
 * 除非符合Noark许可协议，否则不得使用该文件，您可以下载许可协议文件：
 *
 *        http://www.noark.xyz/LICENSE
 *
 * 1.未经许可，任何公司及个人不得以任何方式或理由对本框架进行修改、使用和传播;
 * 2.禁止在本项目或任何子项目的基础上发展任何派生版本、修改版本或第三方版本;
 * 3.无论你对源代码做出任何修改和改进，版权都归Noark研发团队所有，我们保留所有权利;
 * 4.凡侵犯Noark版权等知识产权的，必依法追究其法律责任，特此郑重法律声明！
 */
package xyz.noark.network.rpc;

import xyz.noark.network.codec.rpc.RpcPacket;
import xyz.noark.network.rpc.stub.RpcStub;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * RPC请求存根管理器
 *
 * @author 小流氓[176543888@qq.com]
 * @since 4.0
 */
public class RpcStubManager {
    private static final RpcStubManager instance = new RpcStubManager();
    private static final AtomicInteger reqId = new AtomicInteger(0);
    private final ConcurrentMap<Integer, RpcStub> stubMap = new ConcurrentHashMap<>();

    public static RpcStubManager getInstance() {
        return instance;
    }

    public int nextReqId() {
        return reqId.incrementAndGet();
    }

    public <T> void bindStub(Integer reqId, RpcStub stub) {
        stubMap.put(reqId, stub);
    }

    public void removeStub(Integer reqId) {
        stubMap.remove(reqId);
    }

    public void onResponse(RpcPacket packet) {
        RpcStub stub = stubMap.get(packet.getReqId());
        // 存根已消失，响应迟到，丢了就行
        if (stub == null) {
            return;
        }
        // 响应回来了
        stub.onResponse(packet);
    }
}